package cmd

import (
	"ai-cli/handler"
	"fmt"
	"github.com/spf13/cobra"
	"github.com/spf13/viper"
)

var rootCmd = &cobra.Command{
	Use:   "aicmd",
	Short: "一个 CMD 工具",
	Long:  "aicmd 是一个支持多命令的 AI 工具",

	Args: cobra.ArbitraryArgs,

	Run: func(cmd *cobra.Command, args []string) {
		if len(args) == 0 {
			fmt.Println("-------------------------------")
			fmt.Println("aicmd setKey '' \t 设置api接口")
			fmt.Println("aicmd info \t 查看配置文件")
			fmt.Println("-------------------------------")
			fmt.Println("欢迎使用 aicmd，这是一个 AI 工具！")
			fmt.Println("当前版本 :", viper.GetString("version"))
		} else {
			text := args[0]
			handler.AiCmd(text)
		}
	},
}

func Execute() {
	if err := rootCmd.Execute(); err != nil {
		fmt.Println(err)
	}
}
